package algorithm_demo.demo02;

/**
 * 前缀树
 * 1. 单个字符串中，字符串从前到后的加到一棵多叉树上
 * 2. 字符串放在路上，节点上有专属的数据项（常见的是pass和end值）
 * 3. 所有样本都这样添加，如果没有路就新建，如有路就复用
 * 4. 沿途节点的pass值增加1，每个字符串结束时来到的节点end值增加1
 * 可以完成前缀相关的查询
 *
 * @author Api
 * @date 2023/4/2 16:56
 */
public class Code07_TrieTree {

    static class Node {
        public int pass;
        public int end;

        public Node[] nexts;

        public Node() {
            this.pass = 0;
            this.end = 0;
            //这里设置26表示26个字符a-z小写
            //0 -> a
            //1 -> b
            //2 -> c
            //......
            //25 -> z
            //nexts[i] == null i方向的路不存在
            //nexts[i] != null i方向的路存在
            this.nexts = new Node[26];
        }
    }

    static class Trie1 {
        private final Node root;

        public Trie1() {
            root = new Node();
        }

        public void insert(String word) {
            if (word == null) {
                return;
            }
            char[] str = word.toCharArray();
            Node node = root;
            node.pass++;
            int path = 0;
            //从左往右遍历字符
            for (int i = 0; i < str.length; i++) {
                //由字符，对应成走哪条路
                path = str[i] - 'a';
                //检查这个方向上的路是不是空的
                if (node.nexts[path] == null) {
                    node.nexts[path] = new Node();
                }
                node = node.nexts[path];
                node.pass++;
            }
            node.end++;
        }

        //word这个单词之前加入过几次
        public int search(String word) {
            if (word == null) {
                return 0;
            }
            char[] str = word.toCharArray();
            Node node = root;
            int index = 0;
            for (int i = 0; i < str.length; i++) {
                index = str[i] - 'a';
                if (node.nexts[index] == null) {
                    return 0;
                }
                node = node.nexts[index];
            }
            return node.end;
        }

        //所有加入的字符串中，有几个是以pre这个字符串作为前缀的
        public int prefixNumber(String pre) {
            if (pre == null) {
                return 0;
            }
            char[] str = pre.toCharArray();
            Node node = root;
            int index = 0;
            for (int i = 0; i < str.length; i++) {
                index = str[i] - 'a';
                if (node.nexts[index] == null){
                    return 0;
                }
                node = node.nexts[index];
            }
            return node.pass;
        }

        public void delete(String word){
            if (search(word) != 0){
                char[] str = word.toCharArray();
                Node node = root;
                node.pass--;
                int index = 0;
                for(int i = 0; i < str.length; i++){
                    index = str[i] - 'a';
                    if (--node.nexts[index].pass == 0){
                        node.nexts[index] = null;
                        return;
                    }
                    node = node.nexts[index];
                }
                node.end--;
            }
        }
    }
}
